Fix testsuite

Signed-off-by: Michal Čihař <michal@cihar.com>
Missing import

Signed-off-by: Michal Čihař <michal@cihar.com>
Simplify condition, we already check for that above

Signed-off-by: Michal Čihař <michal@cihar.com>