diff --git a/src/main/java/net/halbear/hvf/Util/VehicleData/EngineCategories.java b/src/main/java/net/halbear/hvf/Util/VehicleData/EngineCategories.java
index 74666ca..078bc65 100644
--- a/src/main/java/net/halbear/hvf/Util/VehicleData/EngineCategories.java
+++ b/src/main/java/net/halbear/hvf/Util/VehicleData/EngineCategories.java
@@ -3,10 +3,11 @@ package net.halbear.hvf.Util.VehicleData;
 public enum EngineCategories {
     Aircraft_Propeller,
     Aircraft_Jet_Engine,
-    Ship_Propeller,
     Car_Petrol_Engine,
     Car_Diesel_Engine,
     Steam_Ship_Engine,
     Steam_Locomotive,
+    Diesel_Train_Engine,
+    Ship_Diesel_Engine,
     Rocket_Engine
 }