Changed import to support Django 1.6