bug fix...