Removed figure size in test case so hopefully travis will build
correctly