trying to pass the unit tests