fix small bugs
fix bug