Add a HACK to make the @ui @wip hook work