diff --git a/arithmetic/tests/test_operations.py b/arithmetic/tests/test_operations.py
index b555a93..a7eca40 100644
--- a/arithmetic/tests/test_operations.py
+++ b/arithmetic/tests/test_operations.py
@@ -1,5 +1,9 @@
-from arithmetic.operations import add
+from arithmetic.operations import add, sub
 
 
 def test_add():
     assert add(5, 6) == 11
+
+
+def test_sub():
+    assert sub(5, 2) == 3
\ No newline at end of file