Fixing a minor bug that was appearing