Implement query-fn for resettable layer