Correct. Will add tests for that.