more unit tests! How exciting!