Missing dependency

Started beefing up the generator and noticed a dependency missing.