fixed return statement type