Missing import fix