Fixed bug in test_problems