Bug fix for math return type.