bug fix

bug fix to previous commit