diff --git a/test_main.py b/test_main.py
index 2524e4c..5d77c40 100644
--- a/test_main.py
+++ b/test_main.py
@@ -1,3 +1,6 @@
 import numpy as np
 import pandas as pd
+import pytest
+
+
 print("Hello world")