modified the test in the right way