Added a little fix to make this thing compile