Fixed subsequent test failure due to checks on the backend.
Fixed subsequent test failure due to checks on the backend.

balavenkatesan/yellowbrick,jkeung/yellowbrick,DistrictDataLabs/yellowbrick,pdamodaran/yellowbrick