fixed test so build won't break