fixed test not returning mask