Fix. unittest for main.controller