fixing just a little bit