Couple of changes to get sauce labs working