Update for newer pygmalion