Minor code change to match existing methods