Fix missed bit of bad logic