fix small issues