Update syntax to match rest of code