Fixed tests in answer