Updated the test to be right