Fixes import error that broke build.