Update tests for code changes