Updated the code to work with Django 1.5.1.