--- a/CustomCodeGeneratorOrRefactoringTestCaseCodeGenerator.st	Sat Jan 31 19:29:49 2015 +0100
+++ b/CustomCodeGeneratorOrRefactoringTestCaseCodeGenerator.st	Sat Jan 31 23:28:34 2015 +0100
@@ -62,11 +62,16 @@
 !CustomCodeGeneratorOrRefactoringTestCaseCodeGenerator methodsFor:'executing - private'!
 
 generateTestCaseCodeFor:aTestClass forClassUnderTest:aClass 
-    | source className |
+    | source className package |
 
     super generateTestCaseCodeFor:aTestClass forClassUnderTest:aClass.
     className := aClass theNonMetaClass name.
 
+    package := PackageId noProjectID.
+    samePackageAsTestedClass ifTrue: [ 
+        package := aClass package
+    ].
+
     source := 'test_code_generated
     | expectedSource |
 
@@ -81,6 +86,7 @@
 
         protocol: 'tests';
         source: source;
+        package: package;
         compile.
 
     model createMethod
@@ -90,10 +96,11 @@
         source: 'generatorOrRefactoring
     ^ `className new';
         replace: '`className' with: className asString;
+        package: package;
         compile
 
     "Created: / 16-09-2014 / 10:40:58 / Jan Vrany <jan.vrany@fit.cvut.cz>"
-    "Modified: / 30-10-2014 / 21:16:38 / Jakub Nesveda <nesvejak@fit.cvut.cz>"
+    "Modified: / 31-01-2015 / 18:39:38 / Jakub Nesveda <nesvejak@fit.cvut.cz>"
 ! !
 
 !CustomCodeGeneratorOrRefactoringTestCaseCodeGenerator class methodsFor:'documentation'!