Refactored test to work with the new name