Don't use deprecated argument in tests