Fix non-compiling code ^_^