Ajay | Adding missing import and fixing exception logging