Fixed single broken test