TST: Use correct expectations for AS in tests