Fixing test to not take so long