just fixed return code