diff --git a/NzbDrone.Core.Test/MediaFileTests/EpisodeImportTests/NotSampleSpecificationFixture.cs b/NzbDrone.Core.Test/MediaFileTests/EpisodeImportTests/NotSampleSpecificationFixture.cs
index c55fc69be..f4b282eaa 100644
--- a/NzbDrone.Core.Test/MediaFileTests/EpisodeImportTests/NotSampleSpecificationFixture.cs
+++ b/NzbDrone.Core.Test/MediaFileTests/EpisodeImportTests/NotSampleSpecificationFixture.cs
@@ -55,9 +55,7 @@ namespace NzbDrone.Core.Test.MediaFileTests.EpisodeImportTests
 
         private void WithFileSize(long size)
         {
-            Mocker.GetMock<IDiskProvider>()
-                  .Setup(s => s.GetFileSize(It.IsAny<String>()))
-                  .Returns(size);
+            _localEpisode.Size = size;
         }
 
         private void WithLength(int minutes)