Fix issue with host and missing import in python impl

Fix issue with host and missing import in python impl