small changes
fix