fixing test stuff