Fixed unit test
Whoops