From 0f97930f7918b8deebbb8e3cb832217288ba8728 Mon Sep 17 00:00:00 2001
From: Felix Cravic <themode@outlook.fr>
Date: Sun, 6 Dec 2020 16:52:04 +0100
Subject: [PATCH] Fix player gravity

---
 src/main/java/net/minestom/server/entity/Entity.java            | 2 +-
 .../minestom/server/network/player/NettyPlayerConnection.java   | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/main/java/net/minestom/server/entity/Entity.java b/src/main/java/net/minestom/server/entity/Entity.java
index bb5d8d7af..0ded5851e 100644
--- a/src/main/java/net/minestom/server/entity/Entity.java
+++ b/src/main/java/net/minestom/server/entity/Entity.java
@@ -481,7 +481,7 @@ public abstract class Entity implements Viewable, EventHandler, DataContainer, P
                 // Gravity
                 {
                     // Cache the number of "gravity tick"
-                    if (!isOnGround()) {
+                    if (!onGround) {
                         gravityTickCount++;
                     } else {
                         gravityTickCount = 0;
diff --git a/src/main/java/net/minestom/server/network/player/NettyPlayerConnection.java b/src/main/java/net/minestom/server/network/player/NettyPlayerConnection.java
index d2c949821..7b0ecaefc 100644
--- a/src/main/java/net/minestom/server/network/player/NettyPlayerConnection.java
+++ b/src/main/java/net/minestom/server/network/player/NettyPlayerConnection.java
@@ -143,6 +143,7 @@ public class NettyPlayerConnection extends PlayerConnection {
         }
     }
 
+    @NotNull
     public ChannelFuture write(@NotNull Object message) {
         if (MinecraftServer.shouldProcessNettyErrors()) {
             return channel.write(message).addListener(future -> {
@@ -155,6 +156,7 @@ public class NettyPlayerConnection extends PlayerConnection {
         }
     }
 
+    @NotNull
     public ChannelFuture writeAndFlush(@NotNull Object message) {
         if (MinecraftServer.shouldProcessNettyErrors()) {
             return channel.writeAndFlush(message).addListener(future -> {