FIX BUGS