Small fix with warnings