FIX latest bugs