Fix no method error