Tiny bug fixes

Everything works on the new UI