fix @muan's unit test ;)