Fixing version unit tests