Fixed references in the tests