Update tests to work with new vector class