trying to fix a little bug