Bug/gomega fix