Changed checks to actual assert statements