Fixing a little bug