changed tests slightly