fixed the test to work as expected