From 517c41b637c1645139aea56c862a29015d3fe3fc Mon Sep 17 00:00:00 2001
From: Terpo <mail@terpo.org>
Date: Sat, 14 Mar 2020 17:56:45 +0100
Subject: [PATCH] fix missing entity type classification #92

---
 src/main/java/com/simibubi/create/AllEntities.java | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/src/main/java/com/simibubi/create/AllEntities.java b/src/main/java/com/simibubi/create/AllEntities.java
index 0ab3eeb39..7da80aa5f 100644
--- a/src/main/java/com/simibubi/create/AllEntities.java
+++ b/src/main/java/com/simibubi/create/AllEntities.java
@@ -19,9 +19,10 @@ import net.minecraftforge.fml.client.registry.RenderingRegistry;
 
 public enum AllEntities {
 
-	CONTRAPTION(ContraptionEntity::new, 30, 3, true, ContraptionEntity::build),
-	STATIONARY_CONTRAPTION(ContraptionEntity::new, 30, 40, false, ContraptionEntity::build),
-
+	CONTRAPTION(ContraptionEntity::new, EntityClassification.MISC, 30, 3, true,
+			ContraptionEntity::build), 
+	STATIONARY_CONTRAPTION(ContraptionEntity::new, EntityClassification.MISC, 30, 40,
+					false, ContraptionEntity::build),
 	;
 
 	private IFactory<?> factory;
@@ -33,9 +34,11 @@ public enum AllEntities {
 
 	public EntityType<? extends Entity> type;
 
-	private AllEntities(IFactory<?> factory, int range, int updateFrequency, boolean sendVelocity,
+	private AllEntities(IFactory<?> factory, EntityClassification group, int range, int updateFrequency,
+			boolean sendVelocity,
 			Function<EntityType.Builder<? extends Entity>, EntityType.Builder<? extends Entity>> propertyBuilder) {
 		this.factory = factory;
+		this.group = group;
 		this.range = range;
 		this.updateFrequency = updateFrequency;
 		this.sendVelocity = sendVelocity;