Fixed my previous commit. Works now.