Fix line code on controller