Fix build issue from 5edb708

Motivation:
Commit 5edb708 introduced a build failure.

Modifications:
Add missing imports so build can complete.

Result:
Compilation error no more.