diff --git a/groupConnectivity.pyx b/groupConnectivity.pyx
index 4d440bf4b8bc3b4173af279ebacf2d98962197b4..d7043b8e71fa1d3d4123bad812be858eef859498 100644
--- a/groupConnectivity.pyx
+++ b/groupConnectivity.pyx
@@ -1,3 +1,5 @@
+include "cysignals/signals.pxi"
+
 from libcpp.vector cimport vector
 from libcpp cimport bool
 from libcpp.utility cimport pair
@@ -138,7 +140,9 @@ def testGroupConnectivity(G, group = "Z4", getClasses = False,
     print "Double subdivided edges: ", doubleSubdivisions
 
   tester.init(G.num_edges(), spanningTreeEdges, twoCuts, doubleSubdivisions, elemCycles)
+  sig_on()
   ret = tester.run()
+  sig_off()
 
   if getClasses:
     classes = tester.getClasses()