Small change in some testing code.
Adding some utils.