From 28304b1a281ef3b331e6e27ccfaf0bebfed8e7dd Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Florian=20N=C3=BCcke?= <florian@nuecke.de>
Date: Thu, 27 Jan 2022 11:15:38 +0100
Subject: [PATCH] Make some more block entities use custom base class for
 consistency.

---
 .../cil/oc2/common/blockentity/CreativeEnergyBlockEntity.java  | 2 +-
 .../oc2/common/blockentity/RedstoneInterfaceBlockEntity.java   | 3 +--
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/src/main/java/li/cil/oc2/common/blockentity/CreativeEnergyBlockEntity.java b/src/main/java/li/cil/oc2/common/blockentity/CreativeEnergyBlockEntity.java
index 854e5e66..176444b9 100644
--- a/src/main/java/li/cil/oc2/common/blockentity/CreativeEnergyBlockEntity.java
+++ b/src/main/java/li/cil/oc2/common/blockentity/CreativeEnergyBlockEntity.java
@@ -8,7 +8,7 @@ import net.minecraft.world.level.Level;
 import net.minecraft.world.level.block.entity.BlockEntity;
 import net.minecraft.world.level.block.state.BlockState;
 
-public final class CreativeEnergyBlockEntity extends BlockEntity {
+public final class CreativeEnergyBlockEntity extends ModBlockEntity {
     private final Direction[] SIDES = Direction.values();
 
     ///////////////////////////////////////////////////////////////////
diff --git a/src/main/java/li/cil/oc2/common/blockentity/RedstoneInterfaceBlockEntity.java b/src/main/java/li/cil/oc2/common/blockentity/RedstoneInterfaceBlockEntity.java
index 5aa83434..44b070b3 100644
--- a/src/main/java/li/cil/oc2/common/blockentity/RedstoneInterfaceBlockEntity.java
+++ b/src/main/java/li/cil/oc2/common/blockentity/RedstoneInterfaceBlockEntity.java
@@ -12,7 +12,6 @@ import net.minecraft.core.Direction;
 import net.minecraft.nbt.CompoundTag;
 import net.minecraft.util.Mth;
 import net.minecraft.world.level.ChunkPos;
-import net.minecraft.world.level.block.entity.BlockEntity;
 import net.minecraft.world.level.block.state.BlockState;
 
 import javax.annotation.Nullable;
@@ -20,7 +19,7 @@ import java.util.Collection;
 
 import static java.util.Collections.singletonList;
 
-public final class RedstoneInterfaceBlockEntity extends BlockEntity implements NamedDevice, DocumentedDevice {
+public final class RedstoneInterfaceBlockEntity extends ModBlockEntity implements NamedDevice, DocumentedDevice {
     private static final String OUTPUT_TAG_NAME = "output";
 
     private static final String GET_REDSTONE_INPUT = "getRedstoneInput";