diff --git a/Source/StevesUEHelpers/Private/StevesUI/InputImage.cpp b/Source/StevesUEHelpers/Private/StevesUI/InputImage.cpp
index 44794c0..a6ab808 100644
--- a/Source/StevesUEHelpers/Private/StevesUI/InputImage.cpp
+++ b/Source/StevesUEHelpers/Private/StevesUI/InputImage.cpp
@@ -1,9 +1,8 @@
 #include "StevesUI/InputImage.h"
-#include "StevesUI/KeySprite.h"
 #include "StevesGameSubsystem.h"
 #include "StevesUEHelpers.h"
 #include "Blueprint/WidgetTree.h"
-#include "Engine/AssetManager.h"
+#include "InputAction.h"
 
 TSharedRef<SWidget> UInputImage::RebuildWidget()
 {