diff --git a/test/test_model_modules/test_model_class.py b/test/test_model_modules/test_model_class.py
index cbff4cec6c5b002c3166954880e1008e7f4d7ae3..b05fd990c79b881124fa86fcccaeb4d9c1976d5b 100644
--- a/test/test_model_modules/test_model_class.py
+++ b/test/test_model_modules/test_model_class.py
@@ -1,14 +1,14 @@
 import keras
 import pytest
 
-from mlair.model_modules.model_class import MyPaperModel
+from mlair.model_modules.model_class import IntelliO3_ts_architecture
 
 
-class TestMyPaperModel:
+class TestIntelliO3_ts_architecture:
 
     @pytest.fixture
     def mpm(self):
-        return MyPaperModel(input_shape=[(7, 1, 9)], output_shape=[(4,)])
+        return IntelliO3_ts_architecture(input_shape=[(7, 1, 9)], output_shape=[(4,)])
 
     def test_init(self, mpm):
         # check if loss number of loss functions fit to model outputs