Fix bug in v1