Use expect instead of should