Fix import for Statecontroller