Small amount of code cleanup.
Fixed the build.