Updated CG test to work in TravisCI