Update TestController to match current input style