Changes code to use matrix::CheckAugmentable() instead of math::CheckAugmentable().