Small fix with exit