Small fix to ds check