Fixed method return type error