Fix live tests