fix(test): Make tests working again.