[BUGFIX] Fixed unit test.