Test not reflecting change in code