fixing two tests

freestyle fix