fix code black prettyprint bug