Fix for missing ui include