Minor fix to imports in new stackdriver module