code review fixes

https://github.com/apache/airflow/commit/5ae96a15fd597ffdc42371156753f8045ac260e6