fixed error in code

fixed