Minor Unit Test Change