fixed imports issue with package setup