Fix import grouping