small change to return calculation