Fix bug in D2 handling