minor bug fix
minor bug fix...