updating tests for new model