fixed ut code..to allow for running of unit test.