GUI integration fix

Forgot about UI callback stuff
more integration