Fixed trivial unit test