tests: add missing import

Signed-off-by: Richard Marko <rmarko@fedoraproject.org>