[FIX] Bugs