Updated code for Rocket support.