Fix using wrong models