Fix elastic search unit test