Fix oversight in last commit