add mock import support