diff --git a/module/ui/scroll.py b/module/ui/scroll.py
index a1e38e0b8..a896f657f 100644
--- a/module/ui/scroll.py
+++ b/module/ui/scroll.py
@@ -1,7 +1,7 @@
 import numpy as np
 
 from module.base.base import ModuleBase
-from module.base.button import Button
+from module.base.button import Button, ButtonWrapper
 from module.base.timer import Timer
 from module.base.utils import color_similarity_2d, random_rectangle_point
 from module.logger import logger
@@ -21,7 +21,7 @@ class Scroll:
             is_vertical (bool): True if vertical, false if horizontal.
             name (str):
         """
-        if isinstance(area, Button):
+        if isinstance(area, (Button, ButtonWrapper)):
             # name = area.name
             area = area.area
         self.area = area
@@ -126,10 +126,14 @@ class Scroll:
             random_range (tuple(int, float)):
             distance_check (bool): Whether to drop short swipes
             skip_first_screenshot:
+
+        Returns:
+            bool: If dragged.
         """
         logger.info(f'{self.name} set to {position}')
         self.drag_interval.clear()
         self.drag_timeout.reset()
+        dragged = 0
         if position <= self.edge_threshold:
             random_range = np.subtract(0, self.edge_add)
         if position >= 1 - self.edge_threshold:
@@ -158,6 +162,9 @@ class Scroll:
                 p2 = random_rectangle_point(self.position_to_screen(position, random_range=random_range), n=1)
                 main.device.swipe(p1, p2, name=self.name, distance_check=distance_check)
                 self.drag_interval.reset()
+                dragged += 1
+
+        return dragged
 
     def set_top(self, main, random_range=(-0.05, 0.05), skip_first_screenshot=True):
         return self.set(0.00, main=main, random_range=random_range, skip_first_screenshot=skip_first_screenshot)