diff --git a/pkg/generator_test.go b/pkg/generator_test.go
index 0685be4c..ea0b3456 100644
--- a/pkg/generator_test.go
+++ b/pkg/generator_test.go
@@ -2424,7 +2424,7 @@ import mock "github.com/stretchr/testify/mock"
 	s.checkPrologueGeneration(generator, expected)
 }
 
-func (s *GeneratorSuite) TestInternalPackagePrologue() {
+func (s *GeneratorSuite) TestReplaceTypePackagePrologue() {
 	expected := `package mocks
 
 import baz "github.com/vektra/mockery/v2/pkg/fixtures/example_project/baz"
@@ -2443,7 +2443,7 @@ import mock "github.com/stretchr/testify/mock"
 	s.checkPrologueGeneration(generator, expected)
 }
 
-func (s *GeneratorSuite) TestInternalPackage() {
+func (s *GeneratorSuite) TestReplaceTypePackage() {
 	expected := `// Foo is an autogenerated mock type for the Foo type
 type Foo struct {
 	mock.Mock