Missing import in test