Due to the name change of the main module, correct the 'import' in the
unittest file accordingly.