Test (and fix) math module