#define NUM_CACHEPAGES totalram_pages
 #endif
 
-#define DECL_MMSPACE           mm_segment_t __oldfs
-#define MMSPACE_OPEN \
-       do { __oldfs = get_fs(); set_fs(get_ds()); } while (0)
-#define MMSPACE_CLOSE         set_fs(__oldfs)
-
 #endif /* __LINUX_CFS_MEM_H__ */
 
        struct cfs_trace_page   *tage;
        struct cfs_trace_page   *tmp;
        char                    *buf;
+       mm_segment_t __oldfs;
        int rc;
 
-       DECL_MMSPACE;
-
        cfs_tracefile_write_lock();
 
        filp = filp_open(filename, O_CREAT | O_EXCL | O_WRONLY | O_LARGEFILE,
                rc = 0;
                goto close;
        }
+       __oldfs = get_fs();
+       set_fs(get_ds());
 
        /* ok, for now, just write the pages.  in the future we'll be building
         * iobufs with the pages and calling generic_direct_IO
         */
-       MMSPACE_OPEN;
        list_for_each_entry_safe(tage, tmp, &pc.pc_pages, linkage) {
                __LASSERT_TAGE_INVARIANT(tage);
 
                list_del(&tage->linkage);
                cfs_tage_free(tage);
        }
-       MMSPACE_CLOSE;
+       set_fs(__oldfs);
        rc = vfs_fsync(filp, 1);
        if (rc)
                pr_err("sync returns %d\n", rc);
        struct tracefiled_ctl *tctl = arg;
        struct cfs_trace_page *tage;
        struct cfs_trace_page *tmp;
+       mm_segment_t __oldfs;
        struct file *filp;
        char *buf;
        int last_loop = 0;
        int rc;
 
-       DECL_MMSPACE;
-
        /* we're started late enough that we pick up init's fs context */
        /* this is so broken in uml?  what on earth is going on? */
 
                        __LASSERT(list_empty(&pc.pc_pages));
                        goto end_loop;
                }
-
-               MMSPACE_OPEN;
+               __oldfs = get_fs();
+               set_fs(get_ds());
 
                list_for_each_entry_safe(tage, tmp, &pc.pc_pages, linkage) {
                        static loff_t f_pos;
                                break;
                        }
                }
-               MMSPACE_CLOSE;
+               set_fs(__oldfs);
 
                filp_close(filp, NULL);
                put_pages_on_daemon_list(&pc);