Update tests to use reflect.DeepEqual