fix minor bugs

minor mistakes