Fixed issue with the test