A little error in the code