Missed the input side. This is a dirty quick hack.