fix the silly bug