I think you could use `reflect.DeepEqual` here, instead of comparing manually