fix for the UI - it should actually work now