Fix behave tests