Fix code broken by last commit