fixed up a little bug