Fix plotting non-finite values. (See #148)
Fix non-finite handling that the last commit broke.