Fix error caused by misunderstading. Solution from @HerringtonDarkholme. close #1