Figured I should write a test for the new functionality I just added.