diff --git a/src/main/scripts/init b/src/main/scripts/init
index 3a3b681..7283ba4 100755
--- a/src/main/scripts/init
+++ b/src/main/scripts/init
@@ -29,7 +29,9 @@ if [ -z ${DB2UNIT_PATH} ] ; then
 fi
 
 if [ -z ${DB2UNIT_SRC_MAIN_CODE_PATH} ] ; then
- export DB2UNIT_SRC_MAIN_CODE_PATH=${DB2UNIT_PATH}/sql-pl
- export DB2UNIT_SRC_MAIN_SCRIPT_PATH=${DB2UNIT_PATH}
+ DB2UNIT_SRC_MAIN_CODE_PATH=${DB2UNIT_PATH}/sql-pl
+ DB2UNIT_SRC_MAIN_SCRIPT_PATH=${DB2UNIT_PATH}
+ DB2UNIT_SRC_TEST_CODE_PATH=${DB2UNIT_PATH}/sql-pl
+ DB2UNIT_SRC_TEST_SCRIPT_PATH=${DB2UNIT_PATH}
 fi
 
diff --git a/src/main/scripts/init.bat b/src/main/scripts/init.bat
index f637b35..25e36f6 100644
--- a/src/main/scripts/init.bat
+++ b/src/main/scripts/init.bat
@@ -32,5 +32,7 @@ if "%DB2UNIT_PATH%" EQU "" (
 if "%DB2UNIT_SRC_MAIN_CODE_PATH%" EQU "" (
  set DB2UNIT_SRC_MAIN_CODE_PATH=%DB2UNIT_PATH%sql-pl
  set DB2UNIT_SRC_MAIN_SCRIPT_PATH=%DB2UNIT_PATH%
+ set DB2UNIT_SRC_TEST_CODE_PATH=%DB2UNIT_PATH%sql-pl
+ set DB2UNIT_SRC_TEST_SCRIPT_PATH=%DB2UNIT_PATH%
 )
 
diff --git a/src/main/scripts/init.ps1 b/src/main/scripts/init.ps1
index 15045ad..191b03b 100644
--- a/src/main/scripts/init.ps1
+++ b/src/main/scripts/init.ps1
@@ -31,5 +31,7 @@ if ( ! ( Test-Path Variable:\DB2UNIT_PATH ) ) {
 if ( ! ( Test-Path Variable:\DB2UNIT_SRC_MAIN_CODE_PATH ) ) {
  ${Global:DB2UNIT_SRC_MAIN_CODE_PATH}="${DB2UNIT_PATH}\sql-pl"
  ${Global:DB2UNIT_SRC_MAIN_SCRIPT_PATH}="${DB2UNIT_PATH}"
+ ${Global:DB2UNIT_SRC_TEST_CODE_PATH}="${DB2UNIT_PATH}\sql-pl"
+ ${Global:DB2UNIT_SRC_TEST_SCRIPT_PATH}="${DB2UNIT_PATH}"
 }