fix small problems