Fixing @covers UT errors