Replace assert.Contains with gomega assertions