updated tests so no errors