Updated tests failing