Small fix to mastermine code