fixed error in code