A small bug fix.
The bug is related to !814 (merged)
The bug is basically caused by "import math" and line 292