Fix bug v2
Fix bug v2