Fix test for last commit