diff --git a/python/lbmpy_walberla/packinfo.py b/python/lbmpy_walberla/packinfo.py
index 54a4e675526ec1afb91ff24783397d954faadd17..d6a0ab23c6d1a01dc5f0ba9c866781eba4cf2dba 100644
--- a/python/lbmpy_walberla/packinfo.py
+++ b/python/lbmpy_walberla/packinfo.py
@@ -1,4 +1,5 @@
 from collections import defaultdict
+from lbmpy.stencils import get_stencil
 from lbmpy.advanced_streaming.utility import Timestep, get_accessor, get_timesteps
 from lbmpy.advanced_streaming.communication import _extend_dir
 from pystencils.stencil import inverse_direction
@@ -59,10 +60,12 @@ def generate_lb_pack_info(generation_context,
             for comm_dir in comm_directions(comm_direction):
                 common_spec[(comm_dir,)].add(fa.field.center(*fa.index))
 
+    full_stencil = get_stencil('D3Q27') if len(stencil[0]) == 3 else get_stencil('D2Q9')
+
     for t in timesteps:
         spec = common_spec.copy()
         write_accesses = get_accessor(streaming_pattern, t).write(pdf_field, stencil)
-        for comm_dir in stencil:
+        for comm_dir in full_stencil:
             if all(d == 0 for d in comm_dir):
                 continue