fix incorrect import (fix #4)