Updated tests to reflect constructor signature change