adding back in unit tests