Fixing broken test assertion