fixed return statement bug.