Fixed a mistake from previous commit