minor bug fix
modified version