diff --git a/src/Generator.php b/src/Generator.php
index 1fcdbba..6d1a3bd 100644
--- a/src/Generator.php
+++ b/src/Generator.php
@@ -55,6 +55,7 @@ class Generator
     ) {
         $this->config = $config;
         $this->view = $view;
+        $this->output = $output;
 
         // Find the drivers to add to the extra/interfaces
         $this->detectDrivers();