diff --git a/TB6600MotorDriver.cpp b/TB6600MotorDriver.cpp
index 38e5936..3057cc6 100644
--- a/TB6600MotorDriver.cpp
+++ b/TB6600MotorDriver.cpp
@@ -23,10 +23,10 @@ void TB6600MotorDriver::setup() {
 	pinMode(_pulsePin, OUTPUT);
 
 	digitalWrite(_directionPin, LOW);
-	digitalWrite(_enablePin, HIGH);
+	digitalWrite(_enablePin, _enabled ? HIGH : LOW);
 }
 
-void TB6600MotorDriver::setDirection(uint8_t direction) {
+void TB6600MotorDriver::setDirection(bool direction) {
 	if (direction != _direction) {
 		_direction = direction;
 		_setDirection();
diff --git a/TB6600MotorDriver.h b/TB6600MotorDriver.h
index f13c487..8e44cd3 100644
--- a/TB6600MotorDriver.h
+++ b/TB6600MotorDriver.h
@@ -35,10 +35,9 @@ class TB6600MotorDriver {
   TB6600MotorDriver();
   TB6600MotorDriver(uint8_t enablePin, uint8_t directionPin, uint8_t pulsePin, uint32_t usPulse, uint8_t microsteps);
 
-
   void setup();
 
-  void setDirection(uint8_t direction);
+  void setDirection(bool direction);
   void setSpeed(uint16_t rpm);
 
   //full