Fix import for django 1.6