3 files changed, 3 insertions, 6 deletions
diff --git a/internal/api/client/media/mediacreate_test.go b/internal/api/client/media/mediacreate_test.go
index 471be8557..2ffa8b8a1 100644
--- a/internal/api/client/media/mediacreate_test.go
+++ b/internal/api/client/media/mediacreate_test.go
@@ -110,7 +110,7 @@ func (suite *MediaCreateTestSuite) SetupSuite() {
 }
 
 func (suite *MediaCreateTestSuite) TearDownSuite() {
-	if err := suite.db.Stop(context.Background()); err != nil {
+	if err := suite.db.Close(); err != nil {
 		log.Panicf(nil, "error closing db connection: %s", err)
 	}
 	testrig.StopWorkers(&suite.state)
diff --git a/internal/api/client/media/mediaupdate_test.go b/internal/api/client/media/mediaupdate_test.go
index 1af3bcf06..8140f1acc 100644
--- a/internal/api/client/media/mediaupdate_test.go
+++ b/internal/api/client/media/mediaupdate_test.go
@@ -19,7 +19,6 @@ package media_test
 
 import (
 	"bytes"
-	"context"
 	"encoding/json"
 	"fmt"
 	"io/ioutil"
@@ -107,7 +106,7 @@ func (suite *MediaUpdateTestSuite) SetupSuite() {
 }
 
 func (suite *MediaUpdateTestSuite) TearDownSuite() {
-	if err := suite.db.Stop(context.Background()); err != nil {
+	if err := suite.db.Close(); err != nil {
 		log.Panicf(nil, "error closing db connection: %s", err)
 	}
 	testrig.StopWorkers(&suite.state)
diff --git a/internal/api/fileserver/fileserver_test.go b/internal/api/fileserver/fileserver_test.go
index 709458b1a..e57b86082 100644
--- a/internal/api/fileserver/fileserver_test.go
+++ b/internal/api/fileserver/fileserver_test.go
@@ -18,8 +18,6 @@
 package fileserver_test
 
 import (
-	"context"
-
 	"github.com/stretchr/testify/suite"
 	"github.com/superseriousbusiness/gotosocial/internal/api/fileserver"
 	"github.com/superseriousbusiness/gotosocial/internal/db"
@@ -111,7 +109,7 @@ func (suite *FileserverTestSuite) SetupTest() {
 }
 
 func (suite *FileserverTestSuite) TearDownSuite() {
-	if err := suite.db.Stop(context.Background()); err != nil {
+	if err := suite.db.Close(); err != nil {
 		log.Panicf(nil, "error closing db connection: %s", err)
 	}
 	testrig.StopWorkers(&suite.state)