Updated the tests to handle io.Reader key