Added an assertion to the unit test. Seems to actually be working.