This is the fixed bug that is related to my `TODO` comment on the test. Will remove this.