minor bug fix
bug fixes