Just changed the test