A little fix on previous initial test