Small fixes
fix