Quick fix to compile the code