Updated tests to include the new .model approach