chore: Remove TODO comments from test file