Update tests for new home