diff --git a/tests/basic/test_coder.py b/tests/basic/test_coder.py
index 731b821cd..6cd4de871 100644
--- a/tests/basic/test_coder.py
+++ b/tests/basic/test_coder.py
@@ -2,7 +2,7 @@ import os
 import tempfile
 import unittest
 from pathlib import Path
-from unittest.mock import MagicMock
+from unittest.mock import MagicMock, ANY
 
 import git