[FIX] Fix bugs