improve DS mock