Made behavior tests work