Fixed a bug in the imports of the recently-added code.