Small addition to admin.py