Unit test fixed.