Failed test fix