Fix variable name in tests