simplify test
remove unnessisary code