fix missing imports, thanks to Matthias Lehmann for pointing these out.