Modified the test which needed the django_db mark.