fixing some test stuff