Update tests for improved embedding.