Revert mistake from previous commit