Fix one-line import