Fixed a few missing imports
Merge branch 'fix-initialize_realizations' of https://github.com/gem/oq-engine into fix-initialize_realizations