1 files changed, 2 insertions, 1 deletions
diff --git a/TestStage.h b/TestStage.h
index 3398a89..44a1967 100644
--- a/TestStage.h
+++ b/TestStage.h
@@ -4,6 +4,7 @@
 
 #include "ConstantVelocity.h"
 #include "Enemy.h"
+#include "Flash.h"
 #include "Player.h"
 #include "Stage.h"
 
@@ -17,5 +18,5 @@ public:
 private:
     ConstantVelocitySystem m_const;
     std::vector<Enemy> m_enemies;
-    float m_flash;
+    Flash m_flash;
 };