bug fix....