correct unit testing