Remove circular import