#10 Fixed unit test