Fix AE test error in test_peak_directions_thorough

This is a fix for issue 
https://github.com/nipy/dipy/issues/398