diff --git a/database/migrations/2020_10_14_201320_project_ids_to_entities.php b/database/migrations/2020_10_14_201320_project_ids_to_entities.php
index 9d3af7e58eb5..5b61a0a911fe 100644
--- a/database/migrations/2020_10_14_201320_project_ids_to_entities.php
+++ b/database/migrations/2020_10_14_201320_project_ids_to_entities.php
@@ -21,7 +21,7 @@ class ProjectIdsToEntities extends Migration
             $table->longText('fields')->change();
         });
 
-        Schema::table('gateways', function (Blueprint $table) {
+        Schema::table('companies', function (Blueprint $table) {
             $table->boolean('mark_expenses_invoiceable')->default(0);
             $table->boolean('mark_expenses_paid')->default(0);
             $table->enum('use_credits_payment', ['always', 'off', 'optin'])->nullable();
diff --git a/tests/Unit/AutoBillInvoiceTest.php b/tests/Unit/AutoBillInvoiceTest.php
index 5b7e4c89e05c..26b276b25a09 100644
--- a/tests/Unit/AutoBillInvoiceTest.php
+++ b/tests/Unit/AutoBillInvoiceTest.php
@@ -33,7 +33,9 @@ class AutoBillInvoiceTest extends TestCase
 
     public function testAutoBillFunctionality()
     {
-
+        $this->company->use_credits_payment = 'always';
+        $this->company->save();
+        
         $this->assertEquals($this->client->balance, 10);
         $this->assertEquals($this->client->paid_to_date, 0);
         $this->assertEquals($this->client->credit_balance, 10);