1 files changed, 3 insertions, 2 deletions

diff --git a/src/Chunk.h b/src/Chunk.h
index dd21d0e8f..39dc07a9f 100644
--- a/src/Chunk.h
+++ b/src/Chunk.h
@@ -1,6 +1,7 @@
 
 #pragma once
 
+#include "BlockEntities/BlockEntity.h"
 #include "Entities/Entity.h"
 #include "ChunkData.h"
 
@@ -653,10 +654,10 @@ private:
 	void GetThreeRandomNumbers(int & a_X, int & a_Y, int & a_Z, int a_MaxX, int a_MaxY, int a_MaxZ);
 
 	void RemoveBlockEntity(cBlockEntity * a_BlockEntity);
-	void AddBlockEntity   (cBlockEntity * a_BlockEntity);
+	void AddBlockEntity   (OwnedBlockEntity a_BlockEntity);
 
 	/** Add a block entity to the chunk without marking the chunk dirty */
-	void AddBlockEntityClean(cBlockEntity * a_BlockEntity);
+	void AddBlockEntityClean(OwnedBlockEntity a_BlockEntity);
 
 	/** Creates a block entity for each block that needs a block entity and doesn't have one already */
 	void CreateBlockEntities(void);