Fix record unit tests