Fix broken test :-(