FIx a bug