Nit: unnecessary import