fix: bug in test