Fix test before deploy