Added some change in test