fix bugs ~