Fix bug in store_after_celery_class