Fix some integration issues with Restyler

Signed-off-by: Matt White <matthew.white@fclonline.com>