Fixed small bug when compiling