Fix failing tests

[jrmartin]

Tests are failing after latest changes, need to be updated to pass.

Branch from development

[jrmartin]

Hi @SimaoBolota-MetaCell , when you have a chance can you take care of this, not urgent.

[SimaoBolota-MetaCell]

Fixed by #254