bug fix

Oops.