fixed return type error