Fix test setup for travis