Updated tests for new route