Add check parameter to get_output