:bug: Fix a bug in the test code