Fixed a bug
component.ui returns UI instead of Option[UI]