diff --git a/tests/aggregation/tests.py b/tests/aggregation/tests.py
index <HASH>..<HASH> 100644
--- a/tests/aggregation/tests.py
+++ b/tests/aggregation/tests.py
@@ -3,10 +3,10 @@ from __future__ import absolute_import
 import datetime
 from decimal import Decimal
 
-from django.db.models import Q, F, Avg, Max, Min
+from django.db.models import Q, F, Max, Min
 from django.test import TestCase, Approximate
 
-from aggregate_if import Sum, Count
+from aggregate_if import Sum, Count, Avg
 
 from .models import Author, Publisher, Book, Store
Use conditional Avg on tests