Fixed bug due to recently renamed model