fixes faulty import in test