error with probe screen V2

From what I saw in the pull request, my line numbers are different, don't know exactly if this is what you need, so just replacing yours with this might not be a good idea.
I think its better if you just use it as an inspiration.