Fixes predict output