tests: Fix import.