Fixes another critical import error