Would be nice to maybe see a test for this code too.