diff --git a/godog_and_gomega_helpers.go b/godog_and_gomega_helpers.go
index ae90ef9..2bc85c6 100644
--- a/godog_and_gomega_helpers.go
+++ b/godog_and_gomega_helpers.go
@@ -45,8 +45,8 @@ func Atof(in string) float64 {
 	return res
 }
 
-func SetTestDeadlockProtection() {
-	go checkTestTimeout(1 * time.Second)
+func SetTestDeadlockProtection(seconds int) {
+	go checkTestTimeout(time.Duration(seconds) * time.Second)
 }
 func checkTestTimeout(t time.Duration) {
 	<-time.After(t)