Plugin UI - Bug fixes