Fixed a little mistake from last commit..