Added missing import error in try catch block