diff --git a/src/train_cmd.cpp b/src/train_cmd.cpp
index 492f9ff753..6e253f1866 100644
--- a/src/train_cmd.cpp
+++ b/src/train_cmd.cpp
@@ -3981,7 +3981,7 @@ bool TrainController(Train *v, Vehicle *nomove, bool reverse)
 				v->x_pos = gp.x;
 				v->y_pos = gp.y;
 				v->UpdatePosition();
-				if (v->track == TRACK_BIT_WORMHOLE) {
+				if (HasBit(v->gv_flags, GVF_CHUNNEL_BIT)) {
 					/* update the Z position of the vehicle */
 					int old_z = v->UpdateInclination(false, false, true);