Test failure fix