I'd recommend adding some assertions to test the modeling