less of a nit: and add some test coverage :)