cos distance was used, as it used in DNN stuff and mostly everywhere if you working with vectors. Just example of one of the main distances.
WS distance will work here better, but it little harder to describe then cos.
But you are right it will work better here.