small fixes according new UI