diff --git a/kernelcreation.py b/kernelcreation.py
index d9bf1c7d9a0392d1c930dc52b49fce92573da563..1fb4eeef879a7ad1e8a7deba64cb1c8a01e4fa33 100644
--- a/kernelcreation.py
+++ b/kernelcreation.py
@@ -55,6 +55,8 @@ def create_kernel(assignments, target='cpu', data_type="double", iteration_slice
                [0., 4., 4., 4., 0.],
                [0., 0., 0., 0., 0.]])
     """
+    if isinstance(assignments, AssignmentCollection):
+        assignments = assignments.all_assignments
 
     # ----  Normalizing parameters
     split_groups = ()