Try to fix lagging of UI