#357 Changed include for tests