Would it be possible to move those 2 lines to the controller?