static void heuristic_cleanup_workspace_manager(void)
 {
-       btrfs_cleanup_workspace_manager(&heuristic_wsm);
+       btrfs_cleanup_workspace_manager(BTRFS_COMPRESS_NONE);
 }
 
 static struct list_head *heuristic_get_workspace(unsigned int level)
        }
 }
 
-void btrfs_cleanup_workspace_manager(struct workspace_manager *wsman)
+void btrfs_cleanup_workspace_manager(int type)
 {
+       struct workspace_manager *wsman;
        struct list_head *ws;
 
+       wsman = btrfs_compress_op[type]->workspace_manager;
        while (!list_empty(&wsman->idle_ws)) {
                ws = wsman->idle_ws.next;
                list_del(ws);
 
 struct list_head *btrfs_get_workspace(struct workspace_manager *wsm,
                                      unsigned int level);
 void btrfs_put_workspace(struct workspace_manager *wsm, struct list_head *ws);
-void btrfs_cleanup_workspace_manager(struct workspace_manager *wsm);
+void btrfs_cleanup_workspace_manager(int type);
 
 struct btrfs_compress_op {
        void (*cleanup_workspace_manager)(void);
 
 
 static void lzo_cleanup_workspace_manager(void)
 {
-       btrfs_cleanup_workspace_manager(&wsm);
+       btrfs_cleanup_workspace_manager(BTRFS_COMPRESS_LZO);
 }
 
 static struct list_head *lzo_get_workspace(unsigned int level)
 
 
 static void zlib_cleanup_workspace_manager(void)
 {
-       btrfs_cleanup_workspace_manager(&wsm);
+       btrfs_cleanup_workspace_manager(BTRFS_COMPRESS_ZLIB);
 }
 
 static struct list_head *zlib_get_workspace(unsigned int level)