fix to code mistake