diff --git a/src/MemoryServiceProvider.php b/src/MemoryServiceProvider.php
index 8325c4c..29e99c5 100644
--- a/src/MemoryServiceProvider.php
+++ b/src/MemoryServiceProvider.php
@@ -2,7 +2,7 @@
 
 namespace Orchestra\Memory;
 
-use Illuminate\Contracts\Foundation\Application;
+use Illuminate\Contracts\Container\Container;
 use Orchestra\Support\Providers\ServiceProvider;
 
 class MemoryServiceProvider extends ServiceProvider
@@ -14,7 +14,7 @@ class MemoryServiceProvider extends ServiceProvider
      */
     public function register()
     {
-        $this->app->singleton('orchestra.memory', function (Application $app) {
+        $this->app->singleton('orchestra.memory', function (Container $app) {
             $manager = new MemoryManager($app);
             $namespace = $this->hasPackageRepository() ? 'orchestra/memory::' : 'orchestra.memory';