Updated to fix protected method error