fix regression in last commit