[Minor] fixed broken test