Fixed unit test

Opps