diff --git a/src/main/java/net/frozenblock/trailiertales/TrailierTales.java b/src/main/java/net/frozenblock/trailiertales/TrailierTales.java
index c95eff49..b7aa8736 100644
--- a/src/main/java/net/frozenblock/trailiertales/TrailierTales.java
+++ b/src/main/java/net/frozenblock/trailiertales/TrailierTales.java
@@ -89,7 +89,7 @@ public void onInitialize(String modId, ModContainer container) {
 
 		GravityAPI.MODIFICATIONS.register(gravityContext -> {
 			if (gravityContext.entity != null) {
-				if (gravityContext.entity.getInBlockState().getBlock() instanceof EctoplasmBlock) {
+				if (gravityContext.state.getBlock() instanceof EctoplasmBlock) {
 					gravityContext.gravity *= EctoplasmBlock.GRAVITY_SLOWDOWN;
 				}
 			}