Add tests for preserved state routes