Fixed unit test, added needed objects.