fixed import issue when installing via pip