Little bug fix