Function return value fix