diff --git a/database/test.go b/database/test.go
index 7908dfbcac1f1960c82db6b7c8af2c72d2acab66_ZGF0YWJhc2UvdGVzdC5nbw==..5746457e68bcec77f71b69f5eba624437b2d01ff_ZGF0YWJhc2UvdGVzdC5nbw== 100644
--- a/database/test.go
+++ b/database/test.go
@@ -16,7 +16,7 @@
 
 const dbLockID = 15104
 
-func clearDB(t *testing.T, c *sqlx.Conn, dsn string) {
+func clearDB(t testing.TB, c *sqlx.Conn, dsn string) {
 	// Drop all tables, and reinit
 	if _, err := c.ExecContext(context.Background(), "DROP SCHEMA IF EXISTS public CASCADE"); err != nil {
 		t.Fatal(err)
@@ -72,7 +72,7 @@
 
 // GetTestDB creates a db and returns it. It must be closed within the test.
 // If it fails, t.Fatal() is called
-func GetTestDB(t *testing.T, sourceDriver source.Driver) *TestDB {
+func GetTestDB(t testing.TB, sourceDriver source.Driver) *TestDB {
 	var success bool
 
 	dsn := os.Getenv("TEST_DB_DSN")