Update broken test