lambda-client
diff --git a/‎src/main/kotlin/com/lambda/module/Module.kt‎
Lines changed: 4 additions & 6 deletions b/‎src/main/kotlin/com/lambda/module/Module.kt‎
Lines changed: 4 additions & 6 deletions
@@ -18,7 +18,6 @@
 package com.lambda.module
 
 import com.lambda.Lambda.mc
-import com.lambda.Lambda
 import com.lambda.command.LambdaCommand
 import com.lambda.config.AbstractSetting
 import com.lambda.config.Configurable
@@ -41,7 +40,6 @@ import com.lambda.sound.SoundManager.play
 import com.lambda.util.KeyCode
 import com.lambda.util.Mouse
 import com.lambda.util.Nameable
-import javax.swing.ActionMap
 
 /**
  * A [Module] is a feature or tool for the utility mod.
@@ -143,7 +141,7 @@ abstract class Module(
         listen<MouseEvent.Click>(alwaysListen = true) { event ->
             val pressed = event.action == Mouse.Action.Click.ordinal
             val released = event.action == Mouse.Action.Release.ordinal
-            onButtonPress(event.button, pressed, released, mouseButton = true)
+            onButtonPress(event.button, pressed, released)
         }
 
         onEnable { LambdaSound.MODULE_ON.play() }
@@ -157,12 +155,12 @@ abstract class Module(
         listen<ConnectionEvent.Disconnect> { if (autoDisable) disable() }
     }
 
-    private fun onButtonPress(code: Int, pressed: Boolean, released: Boolean, mouseButton: Boolean = false) {
+    private fun onButtonPress(code: Int, pressed: Boolean, released: Boolean) {
         if (mc.options.commandKey.isPressed) return
         if (mc.currentScreen != null) return
-        if (keybind.code == KeyCode.UNBOUND.code) return
         if (code != keybind.code) return
-        toggle()
+        if (pressed) toggle()
+        else if (released && disableOnRelease) disable()
     }
 
     fun enable() {