Fixed import for django 1.6+