fixed it a little more