last small fixes
fix small error