Remove code from custom ui