fix silly bug