updated tests slightly