Fixed a problem with one of the tests