Adding quick class check to spec