X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Flibr2c2%2Fsignal.cpp;h=a858e0dc92b99293eb142f8172f7902d563acb22;hb=dda1e6c8716b6ac70d63a3f6ff95474a8b8b7336;hp=7ccb23874a7feb6c5659556b513ab4ecaea90a25;hpb=54392d65e2053d1eacb4cfcc435f1013993f2973;p=r2c2.git

diff --git a/source/libr2c2/signal.cpp b/source/libr2c2/signal.cpp
index 7ccb238..a858e0d 100644
--- a/source/libr2c2/signal.cpp
+++ b/source/libr2c2/signal.cpp
@@ -163,8 +163,8 @@ void Signal::block_reserved(const Block &b, Train *t)
 	{
 		if(t)
 		{
-			int train_entry = t->get_block_allocator().get_entry_to_block(*block);
-			if(train_entry>=0 && static_cast<unsigned>(train_entry)==entry)
+			const BlockIter &b_iter = t->get_block_allocator().iter_for(b);
+			if(b_iter && b_iter.entry()==entry)
 			{
 				if(train_conn)
 					train_conn.disconnect();