Update controller tests