fixing regression in the last commit