Use expect instead of allow