minor bug fix: