Add unit tests, delete incorrect (probably unnecessary) assertion