Fixed (hopefully) test with an assert