fix base on the comments