can you move this to L28 (just above the `with` block)