Fixed omission from last commit