FIx bugs