Bug fix for missing import and wrongly initialized attribute of class