Minor change to use method from Assumption class