Fixed a unit test

Signed-off-by: Byron Maroney <bmaroney@ualberta.ca>