Still more fixes for the python UI