Correct minor bug impacted from last update