Fix error in code