Fixed deprecationwarning with sqrt_shape.

This one is very minor, but fixes a bug that cased a DeprecationWarning to be raised under recent versions of NumPy (I encountered with 1.9.1).