Add more missing include

1 job for release/21.04 in 15 minutes and 48 seconds (queued for 34 seconds)