fixing code.