Fix test to target the correct element