Fixed some code I missed.