Fixed method bug