diff --git a/assignment_collection/simplificationstrategy.py b/assignment_collection/simplificationstrategy.py
index fad2279544a4a6085fbcd091be50ac1688fbbd38..a9e9d0d61bf7b0a64b4b9b4f258f68393196c279 100644
--- a/assignment_collection/simplificationstrategy.py
+++ b/assignment_collection/simplificationstrategy.py
@@ -113,7 +113,7 @@ class SimplificationStrategy(object):
                 def print_assignment_collection(title, c):
                     text = title
                     if self.restrict_symbols:
-                        text += "\n".join([str(e) for e in c.get(self.restrict_symbols)])
+                        text += "\n".join([str(e) for e in c.new_filtered(self.restrict_symbols).main_assignments])
                     else:
                         text += (" " * 3 + (" " * 3).join(str(c).splitlines(True)))
                     return text
@@ -129,7 +129,8 @@ class SimplificationStrategy(object):
                 def print_assignment_collection(title, c):
                     text = '<h5 style="padding-bottom:10px">%s</h5> <div style="padding-left:20px;">' % (title, )
                     if self.restrict_symbols:
-                        text += "\n".join(["$$" + sp.latex(e) + '$$' for e in c.get(self.restrict_symbols)])
+                        text += "\n".join(["$$" + sp.latex(e) + '$$'
+                                           for e in c.new_filtered(self.restrict_symbols).main_assignments])
                     else:
                         # noinspection PyProtectedMember
                         text += c._repr_html_()