Fixing a little issue