Adjustments to unit_test controller method