Fixed a bug where it doesn't compile