[minor] Added missing diagnostics package