diff --git a/tests/trace_simple.py b/tests/test_trace_simple.py
similarity index 90%
rename from tests/trace_simple.py
rename to tests/test_trace_simple.py
index 7038ac2..859121c 100644
--- a/tests/trace_simple.py
+++ b/tests/test_trace_simple.py
@@ -34,5 +34,5 @@ def forward(self, x):
 config_pipeline = PipelineConfig(input_shape=[batch_size, 3, 32, 32],
                                  output_shape=[batch_size],
                                  data_type="long")
-
-trace = SkippableTracing(nb_gpus=0, model=model, config=config_pipeline)
\ No newline at end of file
+def test_trace_simple():
+    trace = SkippableTracing(nb_gpus=0, model=model, config=config_pipeline)