Bug fixing in tests reader class when returning input data dict