diff --git a/src/tests/Feature/Console/Wallet/GetDiscountTest.php b/src/tests/Feature/Console/Wallet/GetDiscountTest.php
index a5ac029a..61c2906c 100644
--- a/src/tests/Feature/Console/Wallet/GetDiscountTest.php
+++ b/src/tests/Feature/Console/Wallet/GetDiscountTest.php
@@ -1,65 +1,77 @@
 <?php
 
 namespace Tests\Feature\Console\Wallet;
 
 use Illuminate\Support\Facades\Queue;
 use Tests\TestCase;
 
 class GetDiscountTest extends TestCase
 {
     /**
      * {@inheritDoc}
      */
     public function setUp(): void
     {
         parent::setUp();
 
         $this->deleteTestUser('wallets-controller@kolabnow.com');
     }
 
     /**
      * {@inheritDoc}
      */
     public function tearDown(): void
     {
         $this->deleteTestUser('wallets-controller@kolabnow.com');
 
         parent::tearDown();
     }
 
     /**
      * Test command run for a specified wallet
      */
     public function testHandle(): void
     {
         Queue::fake();
 
         $user = $this->getTestUser('wallets-controller@kolabnow.com');
         $wallet = $user->wallets()->first();
 
         // Non-existing wallet
         $code = \Artisan::call("wallet:get-discount 123");
         $output = trim(\Artisan::output());
 
         $this->assertSame(1, $code);
         $this->assertSame("Wallet not found.", $output);
 
         // No discount
         $code = \Artisan::call("wallet:get-discount {$wallet->id}");
         $output = trim(\Artisan::output());
 
         $this->assertSame(0, $code);
         $this->assertSame("No discount on this wallet.", $output);
 
-        $discount = \App\Discount::withObjectTenantContext($user)->where('discount', 100)->first();
+        $code = \Artisan::call("wallet:get-discount {$wallet->id} --int");
+        $output = trim(\Artisan::output());
+
+        $this->assertSame(0, $code);
+        $this->assertSame("0", $output);
+
+        $discount = \App\Discount::withObjectTenantContext($user)->where('discount', 10)->first();
         $wallet->discount()->associate($discount);
         $wallet->save();
 
         // With discount
         $code = \Artisan::call("wallet:get-discount {$wallet->id}");
         $output = trim(\Artisan::output());
 
         $this->assertSame(0, $code);
-        $this->assertSame("100", $output);
+        $this->assertSame("10% [TEST] Test voucher", $output);
+
+        $code = \Artisan::call("wallet:get-discount {$wallet->id} --int");
+        $output = trim(\Artisan::output());
+
+        $this->assertSame(0, $code);
+        $this->assertSame("10", $output);
     }
 }