Fixing Trying to get property of non-object.