Fix UI build