Minor update to the test.
- Now uses source generator.