diff --git a/tests/Feature/Controllers/ImportControllerTest.php b/tests/Feature/Controllers/ImportControllerTest.php
index 9b37820f2b..23f1d47788 100644
--- a/tests/Feature/Controllers/ImportControllerTest.php
+++ b/tests/Feature/Controllers/ImportControllerTest.php
@@ -150,6 +150,7 @@ class ImportControllerTest extends TestCase
         $configurator->shouldReceive('setJob')->once();
         $configurator->shouldReceive('isJobConfigured')->once()->andReturn(false);
         $configurator->shouldReceive('configureJob')->once()->andReturn(false);
+        $configurator->shouldReceive('getWarningMessage')->once()->andReturn('');
 
 
         $this->be($this->user());
diff --git a/tests/Feature/Controllers/TagControllerTest.php b/tests/Feature/Controllers/TagControllerTest.php
index 7e0feca5fb..779713ee06 100644
--- a/tests/Feature/Controllers/TagControllerTest.php
+++ b/tests/Feature/Controllers/TagControllerTest.php
@@ -86,8 +86,6 @@ class TagControllerTest extends TestCase
         $repository   = $this->mock(TagRepositoryInterface::class);
         $journalRepos = $this->mock(JournalRepositoryInterface::class);
         $journalRepos->shouldReceive('first')->once()->andReturn(new TransactionJournal);
-        $repository->shouldReceive('tagAllowAdvance')->once()->andReturn(false);
-        $repository->shouldReceive('tagAllowBalancing')->once()->andReturn(false);
 
         $this->be($this->user());
         $response = $this->get(route('tags.edit', [1]));