fix the newly added test