Fixed typo, added a bit to the test