Added correct import for MATSet