minor bug fix..