Update for latest piston

@bvssvni should this be a pull request or should I just make the commit myself? Not sure how you guys do things.