Fix behaves tests