Fix test error from 1cf10b57e89cfc260047397411f1f83d452f8f00