Set correct return type