fixed test of code