correct unit tests