Changed import to support Django 1.6 version