Fix namespace issue in Controller