fixed test

Minor fixes to allow unit test to function