Updated pit-test plugin