X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fengineer%2Ftrainproperties.cpp;h=d92750e41725fd7d65e7071fbcfefc7e785c5cb1;hb=ca9d278f9472206ad9a01190dcef9f0eb1bcc274;hp=ecc9b2b73e555c718927ea626d540f00744ba6af;hpb=d0af7846e05691d65d8021e46c8f81e8ca05199a;p=r2c2.git

diff --git a/source/engineer/trainproperties.cpp b/source/engineer/trainproperties.cpp
index ecc9b2b..d92750e 100644
--- a/source/engineer/trainproperties.cpp
+++ b/source/engineer/trainproperties.cpp
@@ -90,7 +90,7 @@ TrainProperties::TrainProperties(Engineer &e, Train *t):
 	{
 		ent_addr->set_text(lexical_cast<string>(train->get_address()));
 		ent_name->set_text(train->get_name());
-		if(TrainRouter *router = dynamic_cast<TrainRouter *>(train->get_tagged_ai("router")))
+		if(TrainRouter *router = train->get_ai_of_type<TrainRouter>())
 			drp_priority->set_selected_index(router->get_priority()+2);
 
 		unsigned n_vehicles = train->get_n_vehicles();
@@ -117,7 +117,7 @@ void TrainProperties::on_ok_clicked()
 	}
 
 	train->set_name(ent_name->get_text());
-	if(TrainRouter *router = dynamic_cast<TrainRouter *>(train->get_tagged_ai("router")))
+	if(TrainRouter *router = train->get_ai_of_type<TrainRouter>())
 		router->set_priority(drp_priority->get_selected_index()-2);
 
 	// The locomotive is vehicle 0 so we need to add 1