remove tests of legacy models and fix unit test to pass