fix small issues with test