diff --git a/tests/N98/Magento/ModulesTest.php b/tests/N98/Magento/ModulesTest.php
index 339961dc0..7d89fb091 100644
--- a/tests/N98/Magento/ModulesTest.php
+++ b/tests/N98/Magento/ModulesTest.php
@@ -123,18 +123,11 @@ private function filter($option = null, $value = null)
             ->setMethods(['getOption'])
             ->getMock();
 
-        $i = 0;
-        foreach ($options as $opt => $val) {
-            if (!$val) {
-                continue;
-            }
-
-            $input
-                ->expects(self::at($i++))
-                ->method('getOption')
-                ->with($opt)
-                ->willReturn($val);
-        }
+        $input
+            ->expects(self::any())
+            ->method('getOption')
+            ->with($option)
+            ->willReturn($options[$option]);
 
         return $input;
     }