diff --git a/rail_cmd.c b/rail_cmd.c
--- a/rail_cmd.c
+++ b/rail_cmd.c
@@ -665,7 +665,7 @@ int32 CmdBuildTrainDepot(int x, int y, u
 		d->xy = tile;
 		d->town_index = ClosestTownFromTile(tile, (uint)-1)->index;
 
-		SetSignalsOnBothDir(tile, (p2 & 1) ? 2 : 1);
+		UpdateSignalsOnSegment(tile, p2);
 	}
 
 	return cost + _price.build_train_depot;
@@ -1023,10 +1023,10 @@ static int32 RemoveTrainDepot(TileIndex 
 		return CMD_ERROR;
 
 	if (flags & DC_EXEC) {
-		Track track = TrackdirToTrack(DiagdirToDiagTrackdir(GetRailDepotDirection(tile)));
+		DiagDirection dir = GetRailDepotDirection(tile);
 
 		DoDeleteDepot(tile);
-		SetSignalsOnBothDir(tile, track);
+		UpdateSignalsOnSegment(tile, dir);
 	}
 
 	return _price.remove_train_depot;