FIX : bugs