Fixed a build issue caused by refactoring.