diff --git a/tests/test_augmentations/test_mosaic.py b/tests/test_augmentations/test_mosaic.py
index 516d1b59..183e1e97 100644
--- a/tests/test_augmentations/test_mosaic.py
+++ b/tests/test_augmentations/test_mosaic.py
@@ -26,7 +26,7 @@ def test_bbox_mosaic4():
         mosaic_bbox = bbox_mosaic4(
             bbox, HEIGHT // 2, WIDTH // 2, i, HEIGHT, WIDTH, 0, 0
         )
-        assert pytest.approx(mosaic_bbox, abs=0.5) == (
+        assert pytest.approx(mosaic_bbox, abs=1) == (
             0,
             0,
             WIDTH // 2,
@@ -35,7 +35,6 @@ def test_bbox_mosaic4():
 
 
 def test_keypoint_mosaic4():
-    keypoint = (WIDTH // 2, HEIGHT // 2, 0, 0)
     for i, (w, h) in enumerate(
         [
             (WIDTH // 2, HEIGHT // 2),
@@ -45,9 +44,9 @@ def test_keypoint_mosaic4():
         ]
     ):
         mosaic_keypoint = keypoint_mosaic4(
-            keypoint, HEIGHT // 2, WIDTH // 2, i, HEIGHT, WIDTH, 0, 0
+            (w, h, 0, 0), HEIGHT // 2, WIDTH // 2, i, HEIGHT, WIDTH, 0, 0
         )
-        assert pytest.approx(mosaic_keypoint, abs=0.25) == (w, h, 0, 0)
+        assert pytest.approx(mosaic_keypoint, abs=0.25) == (w * 2, h * 2, 0, 0)
 
 
 def test_Mosaic4():