Updated test according to comments