1 files changed, 9 insertions, 4 deletions
diff --git a/src/T2DLL/T2EquipInfoDialog.h b/src/T2DLL/T2EquipInfoDialog.h
index cc6ce0b..751e222 100644
--- a/src/T2DLL/T2EquipInfoDialog.h
+++ b/src/T2DLL/T2EquipInfoDialog.h
@@ -1,11 +1,16 @@
 #pragma once
 #include "common.h"
+#include "T2Equip.h"
+#include "T2InfoDialog.h"
 
-class T2EquipInfoDialog {
+class AFX_EXT_CLASS T2EquipInfoDialog : public T2InfoDialog {
 public:
-	T2EquipInfoDialog(T2Equip*);
+	T2EquipInfoDialog(T2Equip* inEquip);
 	virtual ~T2EquipInfoDialog();
 
-	T2Equip* GetEquip() const {}
-	T2EquipDef* GetEquipDef() const {}
+	T2Equip* GetEquip() const { return (T2Equip *) mObject; }
+	T2EquipDef* GetEquipDef() const { return mEquipDef; }
+
+protected:
+    T2EquipDef *mEquipDef;
 };