Fix missing include.

There currently is a build break caused by an incorrect include. This fixes that.