Updated to use Matthew's suggestion for the unit test