Fix

Partially fix the bug that the segmentation controller does not display
properly.