Oops... made a mistake by harcoding the return values