1 files changed, 8 insertions, 2 deletions

diff --git a/drivers/net/wireless/wl12xx/ps.c b/drivers/net/wireless/wl12xx/ps.c
index a7a11088dd31..a2bdacdd7e1d 100644
--- a/drivers/net/wireless/wl12xx/ps.c
+++ b/drivers/net/wireless/wl12xx/ps.c
@@ -53,8 +53,11 @@ void wl1271_elp_work(struct work_struct *work)
                 goto out;
 
         wl12xx_for_each_wlvif(wl, wlvif) {
+                if (wlvif->bss_type == BSS_TYPE_AP_BSS)
+                        goto out;
+
                 if (!test_bit(WLVIF_FLAG_PSM, &wlvif->flags) &&
-                    !test_bit(WL1271_FLAG_IDLE, &wl->flags))
+                    test_bit(WLVIF_FLAG_IN_USE, &wlvif->flags))
                         goto out;
         }
 
@@ -78,8 +81,11 @@ void wl1271_ps_elp_sleep(struct wl1271 *wl)
                 return;
 
         wl12xx_for_each_wlvif(wl, wlvif) {
+                if (wlvif->bss_type == BSS_TYPE_AP_BSS)
+                        return;
+
                 if (!test_bit(WLVIF_FLAG_PSM, &wlvif->flags) &&
-                    !test_bit(WL1271_FLAG_IDLE, &wl->flags))
+                    test_bit(WLVIF_FLAG_IN_USE, &wlvif->flags))
                         return;
         }