Found this error in previous test logic and fixed it as well