diff --git a/Plugins/Cog/Source/Cog/Public/CogSubsystem.h b/Plugins/Cog/Source/Cog/Public/CogSubsystem.h
index 401eefc..8fc9631 100644
--- a/Plugins/Cog/Source/Cog/Public/CogSubsystem.h
+++ b/Plugins/Cog/Source/Cog/Public/CogSubsystem.h
@@ -4,6 +4,8 @@
 #include "CogHelper.h"
 #include "CogImguiContext.h"
 #include "CogWindow_Settings.h"
+#include "Components/InputComponent.h"
+
 #include "imgui.h"
 #include "Subsystems/WorldSubsystem.h"
 
diff --git a/Plugins/Cog/Source/Cog/Public/CogWindow.h b/Plugins/Cog/Source/Cog/Public/CogWindow.h
index e844aea..80aab78 100644
--- a/Plugins/Cog/Source/Cog/Public/CogWindow.h
+++ b/Plugins/Cog/Source/Cog/Public/CogWindow.h
@@ -14,6 +14,7 @@ class APlayerController;
 class UCogSubsystem;
 class ULocalPlayer;
 class UWorld;
+class UInputComponent;
 
 class COG_API FCogWindow
 {
diff --git a/Plugins/Cog/Source/Cog/Public/CogWindow_Settings.h b/Plugins/Cog/Source/Cog/Public/CogWindow_Settings.h
index e287f13..a08de74 100644
--- a/Plugins/Cog/Source/Cog/Public/CogWindow_Settings.h
+++ b/Plugins/Cog/Source/Cog/Public/CogWindow_Settings.h
@@ -3,6 +3,7 @@
 #include "CoreMinimal.h"
 #include "CogCommonConfig.h"
 #include "CogWindow.h"
+#include "Framework/Commands/InputChord.h"
 #include "CogWindow_Settings.generated.h"
 
 class UCogEngineConfig_Settings;
diff --git a/Plugins/CogAbility/Source/CogAbility/Public/CogAbilityCheat_Execution_ActivateAbility.h b/Plugins/CogAbility/Source/CogAbility/Public/CogAbilityCheat_Execution_ActivateAbility.h
index e454e73..aa73b45 100644
--- a/Plugins/CogAbility/Source/CogAbility/Public/CogAbilityCheat_Execution_ActivateAbility.h
+++ b/Plugins/CogAbility/Source/CogAbility/Public/CogAbilityCheat_Execution_ActivateAbility.h
@@ -19,9 +19,9 @@ public:
 
     virtual bool GetColor(const FCogWindow& InCallingWindow, FLinearColor& OutColor) const override;
 
-	UPROPERTY(EditAnywhere)
+	UPROPERTY(EditAnywhere, Category="Cog")
     TSubclassOf<UGameplayAbility> Ability;
 
-    UPROPERTY(EditAnywhere)
+	UPROPERTY(EditAnywhere, Category="Cog")
     bool RemoveAbilityOnEnd = true;
 };