fix test stuff