Fix last missing imports

6 jobs for conda-ci in 48 minutes and 36 seconds (queued for 4 seconds)