fixed problem with importing