Fixes for diff graph