fixing mask imports