Fixing bug on test that only shows up on Travis