Fix cleaned up source code

I forgot about it every time I clean it up... :P