fix tests (#1259)
fix tests (#1259)


googledatalab/datalab,craigcitro/datalab,yebrahim/datalab,yebrahim/datalab,googledatalab/datalab,googledatalab/datalab,googledatalab/datalab,yebrahim/datalab,yebrahim/datalab,craigcitro/datalab,googledatalab/datalab,googledatalab/datalab,parthea/datalab,craigcitro/datalab,parthea/datalab,parthea/datalab,yebrahim/datalab,parthea/datalab,parthea/datalab,craigcitro/datalab,craigcitro/datalab,parthea/datalab