fix test

Fix failing test.