Fixed missing method implementation