Fixed a type from Kameron's code