Fix array math test