Updated to get new widget loader