Test failing fix