Fixed a broken test