Fixed new predict mode - without masks