Update unit tests for Active Record models