removed logs
Fixed unit test