unit test working after recent changes