fixing return value for passing test cases