Hook update to enable UI modules