Small fix in TDD