Removed prediction from query method