Add back in Rocketpants specific code