diff --git a/tests/bootstrap.php b/tests/bootstrap.php
index ad1bb10..deb7d0c 100644
--- a/tests/bootstrap.php
+++ b/tests/bootstrap.php
@@ -5,6 +5,6 @@ if (!defined('HHVM_VERSION')) {
     $kernel = \AspectMock\Kernel::getInstance();
     $kernel->init([
         'debug' => true,
-        'includePaths' => [__DIR__ . '/../src/Generator']
+        'includePaths' => [__DIR__ . '/../src']
     ]);
 }