Small fix in Controller test