Fixed bug v2