Fix mask tests