Fixed DeprecationWarning I was guilty of.