fixed it according to the comments