diff --git a/src/Commands/SetTexMapDrawModeCmd.cpp b/src/Commands/SetTexMapDrawModeCmd.cpp
index 2f1d279..380ce93 100644
--- a/src/Commands/SetTexMapDrawModeCmd.cpp
+++ b/src/Commands/SetTexMapDrawModeCmd.cpp
@@ -9,14 +9,14 @@ SetTexMapDrawModeCmd::SetTexMapDrawModeCmd(TextureMappingState * s, int m){
 }
 
 void SetTexMapDrawModeCmd::exec(){
-	ofLogNotice("SetTexMapDrawModeCmd", "undo");
-	_oldMode = 0; // _state->getDrawMode();
-	// _state->setDrawMode(_newMode);
+	ofLogNotice("SetTexMapDrawModeCmd", "exec");
+	_oldMode = _state->getDrawMode();
+	_state->setDrawMode(_newMode);
 }
 
 void SetTexMapDrawModeCmd::undo(){
 	ofLogNotice("SetTexMapDrawModeCmd", "undo");
-	// _state->setDrawMode(_oldMode);
+	_state->setDrawMode(_oldMode);
 }
 
 } // namespace piMapper
diff --git a/src/Commands/SetTexMapDrawModeCmd.h b/src/Commands/SetTexMapDrawModeCmd.h
index 5265962..913cc71 100644
--- a/src/Commands/SetTexMapDrawModeCmd.h
+++ b/src/Commands/SetTexMapDrawModeCmd.h
@@ -10,6 +10,8 @@
 namespace ofx {
 namespace piMapper {
 
+class TextureMappingState;
+
 class SetTexMapDrawModeCmd : public BaseUndoCmd {
 
 	public: