Update test code - no review required