fixes mask dtype -> bool instead of int