X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fengineer%2Ftrainpanel.cpp;h=a0dbbeef3890e345b773344f78c4c8d87ce6d8bd;hb=f74ce147edde73319e55475433adb8e1d87b7201;hp=922e3741e272183abad52382f4255ce08c6fbfe3;hpb=fcaa883538f98bac71ba1a90f98950bb2aa08d88;p=r2c2.git

diff --git a/source/engineer/trainpanel.cpp b/source/engineer/trainpanel.cpp
index 922e374..a0dbbee 100644
--- a/source/engineer/trainpanel.cpp
+++ b/source/engineer/trainpanel.cpp
@@ -262,7 +262,7 @@ void TrainPanel::place(Track *track, unsigned ep)
 {
 	pick_conn.disconnect();
 
-	Block &block = engineer.get_layout().get_block_by_track(*track);
+	Block &block = track->get_block();
 
 	while(1)
 	{
@@ -282,7 +282,7 @@ void TrainPanel::place(Track *track, unsigned ep)
 			Track *next = track->get_links()[ep];
 			ep = next->traverse(next->get_endpoint_by_link(*track), 0);
 			track = next;
-			if(!block.get_tracks().count(track))
+			if(!block.has_track(*track))
 				break;
 		}
 	}