Replace some methods in helper to controller