Changes to sanity test