Shorten test
Remove unneeded method