#define test_opt(sbi, option)  ((sbi)->mount_opt & EROFS_MOUNT_##option)
 
 #ifdef CONFIG_EROFS_FS_ZIP
-#define erofs_workstn_lock(sbi)         xa_lock(&(sbi)->workstn_tree)
-#define erofs_workstn_unlock(sbi)       xa_unlock(&(sbi)->workstn_tree)
-
 /* basic unit of the workstation of a super_block */
 struct erofs_workgroup {
        /* the workgroup index in the workstation */
 
                return err;
 
        sbi = EROFS_SB(sb);
-       erofs_workstn_lock(sbi);
+       xa_lock(&sbi->workstn_tree);
 
        grp = xa_tag_pointer(grp, tag);
 
        /*
         * Bump up reference count before making this workgroup
         * visible to other users in order to avoid potential UAF
-        * without serialized by erofs_workstn_lock.
+        * without serialized by workstn_lock.
         */
        __erofs_workgroup_get(grp);
 
                 */
                __erofs_workgroup_put(grp);
 
-       erofs_workstn_unlock(sbi);
+       xa_unlock(&sbi->workstn_tree);
        radix_tree_preload_end();
        return err;
 }
 
        int i, found;
 repeat:
-       erofs_workstn_lock(sbi);
+       xa_lock(&sbi->workstn_tree);
 
        found = radix_tree_gang_lookup(&sbi->workstn_tree,
                                       batch, first_index, PAGEVEC_SIZE);
                if (unlikely(!--nr_shrink))
                        break;
        }
-       erofs_workstn_unlock(sbi);
+       xa_unlock(&sbi->workstn_tree);
 
        if (i && nr_shrink)
                goto repeat;