Unit tests for controller, theta update fails :(