Remove unused check_output import