fixed code mistake