diff --git a/tests/test_walberla.py b/tests/test_walberla.py
index d0bbd43d5fc22f8fdabb5af7040d8228f864bb52..fd44dde460f4b330692f91fc628ea86b3da2193e 100644
--- a/tests/test_walberla.py
+++ b/tests/test_walberla.py
@@ -21,7 +21,7 @@ from pystencils_autodiff.graph_datahandling import GraphDataHandling
 from pystencils_autodiff.walberla import (
     DefinitionsHeader, FieldAllocation, FlagFieldAllocation, GetParameter,
     ResolveUndefinedSymbols, UniformBlockforestFromConfig, WalberlaMain, WalberlaModule)
-from pystencils_autodiff.wald_und_wiesen_simulation import Simulation
+from pystencils_autodiff.simulation import Simulation
 from pystencils_walberla.cmake_integration import ManualCodeGenerationContext