Fix missing widget

ah, I add it to incorrect place before. I will remove it