Fix after_render