Just a quick code repairing