Mock implementation for dmhandler
Remove unused import