[BUGFIX] Fixed wrong unit test.