More fixes on new extension for the orm