Fixed runtime error. Yay python...