Put plugin assignment in the right place