Adding tests for these 2 cases would be preferable