Fixes ui error