small change to code that might fix a few things