fixing some small issues.