Fixed the logic in my test file.