Fixed the failing test. Failing test was failing because I'm a numpty.