Fix contains method