Fixed bug in my bug fix