Fixed missing method bug