Fix minor logic issue in one of my previous commits