Changed the test, so that the TDD can test prefectly.