Switch to trampolined partition function