thanks for the catch. do you mind also adding unit tests for these?