fixed test stuff