Fix tests (still some deprecieated warnings from DataFrame).