Fixed a error in code