Improved flexibility of get_controller