diff --git a/tests/Feature/Controllers/Budget/AmountControllerTest.php b/tests/Feature/Controllers/Budget/AmountControllerTest.php
index e42ea09c1e..5e1ea56dd3 100644
--- a/tests/Feature/Controllers/Budget/AmountControllerTest.php
+++ b/tests/Feature/Controllers/Budget/AmountControllerTest.php
@@ -149,6 +149,7 @@ class AmountControllerTest extends TestCase
         $repository->shouldReceive('getAvailableBudget')->andReturn('100.123');
         $accountRepos->shouldReceive('setUser');
         $accountRepos->shouldReceive('getAccountsByType')->andReturn(new Collection);
+        $repository->shouldReceive('getAverageAvailable')->andReturn('100.123')->once();
 
         $this->be($this->user());
         $response = $this->get(route('budgets.income.info', ['20170101', '20170131']));
@@ -170,6 +171,7 @@ class AmountControllerTest extends TestCase
         $repository->shouldReceive('getAvailableBudget')->andReturn('100.123');
         $accountRepos->shouldReceive('setUser');
         $accountRepos->shouldReceive('getAccountsByType')->andReturn(new Collection);
+        $repository->shouldReceive('getAverageAvailable')->andReturn('100.123')->once();
 
         $this->be($this->user());
         $this->changeDateRange($this->user(), $range);
diff --git a/tests/Unit/TransactionRules/Triggers/FromAccountStartsTest.php b/tests/Unit/TransactionRules/Triggers/FromAccountStartsTest.php
index a193001169..8c765570e7 100644
--- a/tests/Unit/TransactionRules/Triggers/FromAccountStartsTest.php
+++ b/tests/Unit/TransactionRules/Triggers/FromAccountStartsTest.php
@@ -70,7 +70,6 @@ class FromAccountStartsTest extends TestCase
         $trigger = FromAccountStarts::makeFromStrings(substr($account->name, 0, -3), false);
         $result  = $trigger->triggered($journal);
         $this->assertTrue($result);
-        exit;
     }
 
     /**