forgot to add unit tests back into the checkin