fixing silly bug