Changed sample test to use expect instead of should