Little bug fix...