Fix import problem in mainFrame