diff --git a/tests/TestFunctions.php b/tests/TestFunctions.php
index 205251c..cec660c 100755
--- a/tests/TestFunctions.php
+++ b/tests/TestFunctions.php
@@ -8,8 +8,8 @@ function config($key = null, $default = null)
 }
 
 if (! function_exists('app')) {
-    function app($key = null)
+    function app($key = null, $default = null)
     {
-        return \Torann\LaravelRepository\Test\TestCase::$functions->app($key);
+        return \Torann\LaravelRepository\Test\TestCase::$functions->app($key, $default);
     }
-}
\ No newline at end of file
+}