Add better tests for host model