Add unit tests for other controllers