Updating unit tests to include required params