Attempt to fix test error