Fixed bug

17/17 tests