Added updated test