Fix the cucumber feature

Still not sure about this fix, should research a little bit further