Fixed a bug in one of the tests