Fixing one small issue.. oops