Fixed bug from a test