add logic to handle model tests