diff --git a/contrib/dendrite-demo-i2p/main_test.go b/contrib/dendrite-demo-i2p/main_test.go
index 6eef5454d..bc2ed3934 100644
--- a/contrib/dendrite-demo-i2p/main_test.go
+++ b/contrib/dendrite-demo-i2p/main_test.go
@@ -1,7 +1,10 @@
 package main
 
-/*
 import (
+	"os"
+	"os/signal"
+	"strings"
+	"syscall"
 	"testing"
 )
 
@@ -50,4 +53,4 @@ func TestMain(t *testing.T) {
 		}
 	}
 }
-*/
+
diff --git a/contrib/dendrite-demo-tor/main_test.go b/contrib/dendrite-demo-tor/main_test.go
index ce6c568c1..c90c02908 100644
--- a/contrib/dendrite-demo-tor/main_test.go
+++ b/contrib/dendrite-demo-tor/main_test.go
@@ -1,7 +1,10 @@
 package main
 
-/*
 import (
+	"os"
+	"os/signal"
+	"strings"
+	"syscall"
 	"testing"
 )
 
@@ -50,4 +53,3 @@ func TestMain(t *testing.T) {
 		}
 	}
 }
-*/
\ No newline at end of file