1. UI fix