Move import to a more sensible place