Update test to match template