fixed more unit test stuff