Fixed minor issue introduced after refactoring.