diff --git a/pkg/models/userSource_test.go b/pkg/models/userSource_test.go
index ff8cd95..30b8efa 100644
--- a/pkg/models/userSource_test.go
+++ b/pkg/models/userSource_test.go
@@ -5,9 +5,9 @@ import "testing"
 func TestUserSource_TableName(t *testing.T) {
 	type fields struct {
 		User               User
-		UserID             string
+		UserID             UserID
 		Source             Source
-		SourceID           string
+		SourceID           SourceID
 		ScrapeTimeInterval string
 		AccountUsername    string
 		AccountID          string