Fixed a TypeError, need to stop commiting before testing.