[JRO] Small change to lambda style in test

This is mostly to adhere to the tb style guide in addition to readability