From 1415c9e8dd223f7a87fd8b31041b52f7a0ac93fa Mon Sep 17 00:00:00 2001
From: LmeSzinc <37934724+LmeSzinc@users.noreply.github.com>
Date: Tue, 8 Feb 2022 23:57:27 +0800
Subject: [PATCH] Fix: Wrong arguments of fit_points

---
 module/base/base.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/module/base/base.py b/module/base/base.py
index 92711e00c..2db7d82f7 100644
--- a/module/base/base.py
+++ b/module/base/base.py
@@ -168,9 +168,10 @@ class ModuleBase:
             # Not having enough pixels to match
             return None
 
-        point = fit_points(points, mod=image.shape, encourage=encourage)
+        point = fit_points(points, mod=image_size(image), encourage=encourage)
         point = ensure_int(point + area[:2])
         button_area = area_offset((-encourage, -encourage, encourage, encourage), offset=point)
+        color = get_color(self.device.image, button_area)
         return Button(area=button_area, color=color, button=button_area, name=name)
 
     def interval_reset(self, button):