Update : Fixed bug