minor bug fix...