Use @pytest.mark.flaky decorator instead of @flaky.flaky

Use @pytest.mark.flaky decorator instead of @flaky.flaky #1307