Fixed the custom staff widget.