diff --git a/game/renderer_polygon_3d/polygon_model.cpp b/game/renderer_polygon_3d/polygon_model.cpp
index e49e1a8..e4ac608 100644
--- a/game/renderer_polygon_3d/polygon_model.cpp
+++ b/game/renderer_polygon_3d/polygon_model.cpp
@@ -1,5 +1,7 @@
 #include "polygon_model.hpp"
 
+#include <iostream>
+
 namespace endofthejedi {
     PolygonModel::PolygonModel(const std::string &filename) : m_filename(filename)
     {
diff --git a/game/renderer_polygon_3d/polygon_model.hpp b/game/renderer_polygon_3d/polygon_model.hpp
index 9b67a9a..e3d03d5 100644
--- a/game/renderer_polygon_3d/polygon_model.hpp
+++ b/game/renderer_polygon_3d/polygon_model.hpp
@@ -1,7 +1,7 @@
 #pragma once
 
 #include <string>
-#include <iostream>
+#include <vector>
 
 #include <assimp/Importer.hpp>      // C++ importer interface
 #include <assimp/scene.h>           // Output data structure