using expect instead of should