Fixed @after action handling for controllers