From fd3b42f9437a8bfbdbfd513037014ca702ac3a02 Mon Sep 17 00:00:00 2001
From: Eduardo Sorribas <eduardo@sorribas.org>
Date: Thu, 20 Jun 2013 10:02:06 -0400
Subject: [PATCH] Fix the code styling and add error handling.

---
 index.js | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/index.js b/index.js
index c317818..f7f4ed8 100644
--- a/index.js
+++ b/index.js
@@ -148,11 +148,12 @@ Collection.prototype.group = function(group, callback) {
 };
 
 Collection.prototype.remove = function() {
-	var thiz = this;
+	var self = this;
 	if (arguments[1] == true) { // the justOne parameter
 		var args = arguments;
-		this.find(arguments[0], function(err, docs) {
-			thiz._apply(DRIVER_COLLECTION_PROTO.remove,[docs[0], getCallback(args)]);
+		this.findOne(arguments[0], function(err, doc) {
+			if (err) return getCallback(args)(err);
+			self._apply(DRIVER_COLLECTION_PROTO.remove,[doc, getCallback(args)]);
 		});
 	} else {
 		this._apply(DRIVER_COLLECTION_PROTO.remove, arguments.length === 0 ? [{}, noop] : ensureCallback(arguments));