Fixed bug in Pat's latest feature

It's now awesome!