Switch imports and revert the assert to what it was.