1 files changed, 9 insertions, 1 deletions
diff --git a/tests/units/Model/FileTest.php b/tests/units/Model/FileTest.php
index e7520c89..d1ad7248 100644
--- a/tests/units/Model/FileTest.php
+++ b/tests/units/Model/FileTest.php
@@ -227,7 +227,7 @@ class FileTest extends Base
             ->expects($this->at(1))
             ->method('remove')
             ->with(
-                $this->equalTo('/tmp/foo1')
+                $this->equalTo('thumbnails//tmp/foo2')
             )
             ->will($this->returnValue(true));
 
@@ -235,6 +235,14 @@ class FileTest extends Base
             ->expects($this->at(2))
             ->method('remove')
             ->with(
+                $this->equalTo('/tmp/foo1')
+            )
+            ->will($this->returnValue(true));
+
+        $this->container['objectStorage']
+            ->expects($this->at(3))
+            ->method('remove')
+            ->with(
                 $this->equalTo('/tmp/foo3')
             )
             ->will($this->returnValue(true));