Fix missing import in install and test framework.