Found workaround to include classes in tests