Simplified the test by removing the extra model which doesn't seem to be needed.