Fixed accidentally missing out a test case on answer due to using the wrong variable.