fix typo

Signed-off-by: Michal Toman <mtoman@redhat.com>
add missing import

Signed-off-by: Michal Toman <mtoman@redhat.com>