Fixing the mode to be correct