Reverted a stupid mistake from previous commit