Updated the tests for the vectorization utils