fix for dk unit testing
add fix for dk UATs