diff --git a/tests/LostInTranslationServiceProviderTest.php b/tests/LostInTranslationServiceProviderTest.php
index 84137aa..490804e 100644
--- a/tests/LostInTranslationServiceProviderTest.php
+++ b/tests/LostInTranslationServiceProviderTest.php
@@ -28,7 +28,7 @@ public function testDefaultConfig()
 
         $this->assertArrayHasKey('detect', $config);
         $this->assertEquals([
-            'function' => ['__', 'trans'],
+            'function' => ['__', 'trans', 'trans_choice'],
             'method-function' => [
                 ["app('translator')", 'get'],
             ],
diff --git a/tests/NodeFinder/TranslationFinderTest.php b/tests/NodeFinder/TranslationFinderTest.php
index c61efe5..490c815 100644
--- a/tests/NodeFinder/TranslationFinderTest.php
+++ b/tests/NodeFinder/TranslationFinderTest.php
@@ -39,6 +39,7 @@ public function testFunctionCall()
 
 __('keep');
 trans('keep');
+trans_choice('keep', 10);
 EOD;
 
         $finder = new TranslationFinder(
@@ -47,7 +48,7 @@ public function testFunctionCall()
 
         $nodes = $finder->find($str);
 
-        $this->assertCount(2, $nodes);
+        $this->assertCount(3, $nodes);
     }
 
     public function testStaticCall()