Update tests to match new capitalization