Fix SimpleBatch Test assertions