fix the header import error for v0.1.4