diff --git a/src/main/java/com/simibubi/create/content/curiosities/toolbox/ItemReturnInvWrapper.java b/src/main/java/com/simibubi/create/content/curiosities/toolbox/ItemReturnInvWrapper.java
index 7110649d7..e70e7a36c 100644
--- a/src/main/java/com/simibubi/create/content/curiosities/toolbox/ItemReturnInvWrapper.java
+++ b/src/main/java/com/simibubi/create/content/curiosities/toolbox/ItemReturnInvWrapper.java
@@ -1,15 +1,17 @@
 package com.simibubi.create.content.curiosities.toolbox;
 
 import net.minecraft.world.Container;
+import net.minecraft.world.entity.player.Inventory;
 import net.minecraft.world.item.ItemStack;
 import net.minecraftforge.items.wrapper.InvWrapper;
+import net.minecraftforge.items.wrapper.PlayerMainInvWrapper;
 
 /**
  * For inserting items into a players' inventory anywhere except the hotbar
  */
-public class ItemReturnInvWrapper extends InvWrapper {
+public class ItemReturnInvWrapper extends PlayerMainInvWrapper {
 
-	public ItemReturnInvWrapper(Container inv) {
+	public ItemReturnInvWrapper(Inventory inv) {
 		super(inv);
 	}