Fix build break.
Merge pull request #431 from ImperatorPrime/master

Fix build break.