Fix egregious bug

Thanks Ben