Forgot extra include in test