[MOD]Fix bugs