One small bug. Please merge.

Small bug to a fix I made. Not typical, but needs to be resolved.