Fixes to size classifier tests