Bug: fix MainController