fixed last test a bit