updated accordind to the new test input