big bug fix!