Updated tests so it works now.