minor changes to account for framework changes
Minor change to account for framework change