Fix: Missing include in previous commit