minor fix
minor bug fix