fixing small problems