The assert raises issue even when the calculation is performed properly,
because the assert checks even before `result` is fully
computed. This has been fixed by simply moving the assert line.