GUI integration fix

Forgot about UI callback stuff