fixed return statment