3 files changed, 3 insertions, 6 deletions

diff --git a/drivers/usb/core/hub.c b/drivers/usb/core/hub.c
index d2d648ee8640..a8d879a85d04 100644
--- a/drivers/usb/core/hub.c
+++ b/drivers/usb/core/hub.c
@@ -2808,7 +2808,7 @@ static int hub_thread(void *__unused)
         do {
                 hub_events();
                 wait_event_interruptible(khubd_wait, !list_empty(&hub_event_list)); 
-                try_to_freeze(PF_FREEZE);
+                try_to_freeze();
         } while (!signal_pending(current));
 
         pr_debug ("%s: khubd exiting\n", usbcore_name);
diff --git a/drivers/usb/gadget/file_storage.c b/drivers/usb/gadget/file_storage.c
index 037a7f163822..a9be85103d23 100644
--- a/drivers/usb/gadget/file_storage.c
+++ b/drivers/usb/gadget/file_storage.c
@@ -1554,8 +1554,7 @@ static int sleep_thread(struct fsg_dev *fsg)
         rc = wait_event_interruptible(fsg->thread_wqh,
                         fsg->thread_wakeup_needed);
         fsg->thread_wakeup_needed = 0;
-        if (current->flags & PF_FREEZE)
-                refrigerator(PF_FREEZE);
+        try_to_freeze();
         return (rc ? -EINTR : 0);
 }
 
diff --git a/drivers/usb/storage/usb.c b/drivers/usb/storage/usb.c
index 35c1ca6b5a8e..77e7fc258aa2 100644
--- a/drivers/usb/storage/usb.c
+++ b/drivers/usb/storage/usb.c
@@ -847,10 +847,8 @@ retry:
                 wait_event_interruptible_timeout(us->delay_wait,
                                 test_bit(US_FLIDX_DISCONNECTING, &us->flags),
                                 delay_use * HZ);
-                if (current->flags & PF_FREEZE) {
-                        refrigerator(PF_FREEZE);
+                if (try_to_freeze())
                         goto retry;
-                }
         }
 
         /* If the device is still connected, perform the scanning */