fix import (path changed due to a refactor)