fixing the tests I broke late last night

This plus commit https://github.com/CamDavidsonPilon/lifelines/commit/3f3e90fc47ae45f1cfa1ee49bdb59bec996ced68 will fix master's broken tests. Will merge when :green_apple: