[FIX] Code clean