Fix mixup in last commit