minor bugs fix
minor bug fix