fixing code