Bug fixing: extending input data in multi-label predictions