Fix bug using widget introduced in last commit