Automatically set the host controller on ViewModel returns