Fix controller bug.