diff --git a/Editor/Source/Editor/Windows/SceneEditor/Widgets/ViewportWidget.h b/Editor/Source/Editor/Windows/SceneEditor/Widgets/ViewportWidget.h
index b163e899..f3d6bda5 100644
--- a/Editor/Source/Editor/Windows/SceneEditor/Widgets/ViewportWidget.h
+++ b/Editor/Source/Editor/Windows/SceneEditor/Widgets/ViewportWidget.h
@@ -15,15 +15,28 @@ namespace Nuake
 
 class ViewportWidget : public IEditorWidget
 {
+private:
+	enum class GizmoDrawingModes : char16_t
+	{
+		EditorOnly,
+		Always,
+		None
+	};
+
 private:
 	Ref<Nuake::Viewport> sceneViewport;
 
+	// Gizmo
+	GizmoDrawingModes gizmoDrawingMode;
+
 	ImGuizmo::OPERATION CurrentOperation = ImGuizmo::TRANSLATE;
 	ImGuizmo::MODE CurrentMode = ImGuizmo::WORLD;
 	bool UseSnapping = true;
 	Nuake::Vector3 CurrentSnapping = { 0.05f, 0.05f, 0.05f };
+
 	bool IsControllingCamera = false;
 	bool isHoveringViewport;
+
 public:
 	ViewportWidget(EditorContext& context);
 	~ViewportWidget();