Ensure message stream ends when job completes