Fix for latest breaking changes in Piston