Fix test to use intended variables