Another small fix for my last refactor.