tiny fix in balanced classify