Fix test to look for the right values