Fixing failing test on Travis by simplifying