Little bug correction

:|