--- a/net/minecraft/item/WallOrFloorItem.java
+++ b/net/minecraft/item/WallOrFloorItem.java
@@ -41,4 +41,9 @@
       super.func_195946_a(p_195946_1_, p_195946_2_);
       p_195946_1_.put(this.field_195947_b, p_195946_2_);
    }
+
+   public void removeFromBlockToItemMap(Map<Block, Item> blockToItemMap, Item itemIn) {
+      super.removeFromBlockToItemMap(blockToItemMap, itemIn);
+      blockToItemMap.remove(this.field_195947_b);
+   }
 }
