diff --git a/tests/test_simulator.py b/tests/test_simulator.py
index 0fef977..9d0fd3a 100644
--- a/tests/test_simulator.py
+++ b/tests/test_simulator.py
@@ -1,7 +1,4 @@
-import pytest
-
-import networkx as nx
-
+import numpy as np
 import tsgm
 
 
@@ -23,3 +20,10 @@ def test_sine_cosine_simulator():
     params1 = sin_cosine_sim.params()
     params2 = new_sim.params()
     assert params1.keys() == params2.keys() and params1["max_scale"] == params2["max_scale"] and params1["max_const"] == params2["max_const"]
+
+
+def test_simulator_base():
+    data = tsgm.dataset.Dataset(x=np.ones((3, 2, 1)), y=np.ones(3))
+    s = tsgm.simulator.Simulator(data)
+    cloned = s.clone()
+    assert np.array_equal(s._data.X, cloned._data.X) and np.array_equal(s._data.y, cloned._data.y)