Small change to be able to pass NULL pointer to particle flux