Error Handling
Small fix for dist task