Fix upgrade file tests