fix bug with clear_output and display module