Fix tests to accomodate route change