Fix tests for new route