fix small errors
fix small errors