diff --git a/pystencils_walberla/codegen.py b/pystencils_walberla/codegen.py
index fbe5db3fdc08a2b9d15ec3e881b5e45a6bb3ca28..84c73f070c9be3c8de57669bfba8962b63651bcf 100644
--- a/pystencils_walberla/codegen.py
+++ b/pystencils_walberla/codegen.py
@@ -6,7 +6,7 @@ from typing import Dict, Sequence, Tuple, Optional
 from pystencils import create_staggered_kernel, Field, create_kernel, Assignment, FieldType
 from pystencils.backends.cbackend import get_headers
 from pystencils.backends.simd_instruction_sets import get_supported_instruction_sets
-from pystencils.stencils import offset_to_direction_string, inverse_direction
+from pystencils.stencil import offset_to_direction_string, inverse_direction
 from pystencils_walberla.jinja_filters import add_pystencils_filters_to_jinja_env