graphene: fixing unit tests