Fixed small code mistake