Replace custom assert function with a proper assert module.