fixed the silly bug