fixed it a little