Unbreak Forge build.

Forgot one fix for units.