Update tests according to previous commit