diff --git a/tests/integration/test_components.py b/tests/integration/test_components.py
index 0f61fe4aa..11cd7caee 100644
--- a/tests/integration/test_components.py
+++ b/tests/integration/test_components.py
@@ -270,7 +270,7 @@ def test_populates_devices(self):
 
     def test_updates_mapping(self):
         self.gui.set_active_id("baz")
-        self.controller_mock.update_mapping.called_once_with(target_uinput="baz")
+        self.controller_mock.update_mapping.assert_called_once_with(target_uinput="baz")
 
     def test_selects_correct_target(self):
         self.message_broker.publish(MappingData(target_uinput="baz"))
diff --git a/tests/integration/test_gui.py b/tests/integration/test_gui.py
index 57bb39bd3..7ae71842e 100644
--- a/tests/integration/test_gui.py
+++ b/tests/integration/test_gui.py
@@ -492,10 +492,10 @@ def test_get_color_works(self):
         )
 
         self.assertIsInstance(color, Gdk.RGBA)
-        self.assertNotAlmostEquals(color.red, fallback.red, delta=0.01)
-        self.assertNotAlmostEquals(color.green, fallback.blue, delta=0.01)
-        self.assertNotAlmostEquals(color.blue, fallback.green, delta=0.01)
-        self.assertNotAlmostEquals(color.alpha, fallback.alpha, delta=0.01)
+        self.assertNotAlmostEqual(color.red, fallback.red, delta=0.01)
+        self.assertNotAlmostEqual(color.green, fallback.blue, delta=0.01)
+        self.assertNotAlmostEqual(color.blue, fallback.green, delta=0.01)
+        self.assertNotAlmostEqual(color.alpha, fallback.alpha, delta=0.01)
 
     def _test_color_wont_fallback(self, get_color, fallback):
         color = get_color()