Refactored... changed just one line

I had it this way before, but changed it because something was not working properly. It all works now so I figured I'd bring it back because it looks a bit nicer.