1 files changed, 3 insertions, 3 deletions
diff --git a/src/main/java/me/shedaniel/rei/server/ContainerInfo.java b/src/main/java/me/shedaniel/rei/server/ContainerInfo.java
index 11c90064e..db9b3560d 100644
--- a/src/main/java/me/shedaniel/rei/server/ContainerInfo.java
+++ b/src/main/java/me/shedaniel/rei/server/ContainerInfo.java
@@ -23,10 +23,10 @@
 
 package me.shedaniel.rei.server;
 
-import net.minecraft.container.Container;
+import net.minecraft.screen.ScreenHandler;
 
-public interface ContainerInfo<T extends Container> {
-    Class<? extends Container> getContainerClass();
+public interface ContainerInfo<T extends ScreenHandler> {
+    Class<? extends ScreenHandler> getContainerClass();
     
     int getCraftingResultSlotIndex(T container);