fix small issues...