Fix predictions not being set valid
Fixes #4