1 files changed, 7 insertions, 4 deletions

diff --git a/src/hl2h.cpp b/src/hl2h.cpp
index ca7ee85..703a582 100644
--- a/src/hl2h.cpp
+++ b/src/hl2h.cpp
@@ -909,15 +909,18 @@ static void do_move(int seq, float frametime) {
     note_modified(cmd, flags, vel);
 
   bool melee = feature_active(FEAT_MELEE) && melee_weapon_active(ent);
+  bool gravity = feature_active(FEAT_MELEE) && gravity_gun_active(ent);
+  bool melee_mode = melee || gravity;
 
-  if (!feature_active(FEAT_AIMBOT) || !feature_has_bind(FEAT_AIMBOT) || melee)
+  if (!feature_active(FEAT_AIMBOT) || !feature_has_bind(FEAT_AIMBOT) ||
+      melee_mode)
     aimbot_reset_lock();
   if (!feature_active(FEAT_SILENTAIM) || !feature_has_bind(FEAT_SILENTAIM) ||
-      melee)
+      melee_mode)
     silentaim_reset_lock();
 
-  if (melee) {
-    if (melee_aimbot_apply(cmd))
+  if (melee_mode) {
+    if (melee_aimbot_apply(cmd, gravity && !melee))
       note_modified(cmd, flags, vel);
   } else {
     if (feature_active(FEAT_AIMBOT)) {