Fixed one of the tests for use case 4