Added correct import statements

Reviewed by: Swapna Krishnan <skrishna@redhat.com>