Fixed the tests, Wrong check was added.