Small fix on DL