fixed wrong unit test