Added code to wrong part of test.