[Fixed the broken test]